#include "mystdio.h"
#include <assert.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>

#define FILE_MODE 0666

_FILE * _fopen(const char*filename, const char *flag)
{
    
    if(filename==nullptr||flag==nullptr)
    {
        printf("open failed null point error\n");
        return nullptr;      
    }
    
    //cout<<3<<endl;
    int fd=-1;
    if(strcmp(flag,"w")==0){
        fd=open(filename,O_CREAT|O_WRONLY|O_TRUNC,FILE_MODE);
    }
    else if(strcmp(flag,"r")==0){
        fd=open(filename,O_RDONLY);
    }
    else if(strcmp(flag,"a")==0){
        fd=open(filename,O_CREAT|O_WRONLY|O_APPEND,FILE_MODE);
    }
    else 
        return nullptr;

    if(fd==-1)  return nullptr;
    //cout<<4<<endl;
    _FILE *fp=new(_FILE);
    if(fp==nullptr) return nullptr;
    fp->fileno=fd;
    fp->flag=FLUSH_ALL;
    fp->out_pos=0;
    printf("open file '%s' success , mode = '%s'\n",filename,flag);
    return fp;

}
int _fwrite(_FILE *fp, const char *s, int len)
{
    memcpy(&fp->outbuffer[fp->out_pos],s,len);
    fp->out_pos+=len;
    if(fp->flag==FLUSH_NOW)
    {
        write(fp->fileno,fp->outbuffer,fp->out_pos);
        fp->out_pos=0;
    }
    else if(fp->flag==FLUSH_LINE)
    {
        //暂时简单处理为用户输入了\n
        if(fp->outbuffer[fp->out_pos-1]=='\n')
        {
            write(fp->fileno,fp->outbuffer,fp->out_pos);
            fp->out_pos=0; 
        }
    }
    else if(fp->flag==FLUSH_ALL)
    {
        if(fp->out_pos==SIZE)
        {
            write(fp->fileno,fp->outbuffer,fp->out_pos);
            fp->out_pos=0; 
        }
    }
    else    return -1;
    return len;
    
}
void _fflush(_FILE *fp)
{
    if(fp->out_pos!=0)
    {
        write(fp->fileno,fp->outbuffer,fp->out_pos);
        fp->out_pos=0;
    }
}
void _fclose(_FILE *fp)
{
    if(fp==nullptr) return;
    _fflush(fp);
    close(fp->fileno);
    delete fp; 
}
